package 中等.图;

/**
 * 存在一个 无向图 ，图中有 n 个节点。其中每个节点都有一个介于 0 到 n - 1 之间的
 * 唯一编号。给你一个二维数组 graph ，其中 graph[u] 是一个节点数组，由节点 u 的
 * 邻接节点组成。形式上，对于 graph[u] 中的每个 v ，都存在一条位于节点 u 和节点
 * v 之间的无向边。该无向图同时具有以下属性：
 * 不存在自环（graph[u] 不包含 u）。
 * 不存在平行边（graph[u] 不包含重复值）。
 * 如果 v 在 graph[u] 内，那么 u 也应该在 graph[v] 内（该图是无向图）
 * 这个图可能不是连通图，也就是说两个节点 u 和 v 之间可能不存在一条连通彼此的路径。
 * 二分图 定义：如果能将一个图的节点集合分割成两个独立的子集 A 和 B ，并使图中的
 * 每一条边的两个节点一个来自 A 集合，一个来自 B 集合，就将这个图称为 二分图 。
 * 如果图是二分图，返回 true ；否则，返回 false 。
 * <p>
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode.cn/problems/is-graph-bipartite
 */
public class 判断二分图_785 {

    public static void main(String[] args) {

    }

    /**
     * 染色法
     * graph[0] 代表 0 的邻接点为 graph[0]，0 节点和  graph[0] 节点集合，不能为同一个颜色
     * 这个可以通过 graph[i] 快速定位其邻接点，无需建图
     *
     * @param graph
     * @return
     */
    public boolean isBipartite(int[][] graph) {
        int[] colors = new int[graph.length];

        for (int cur = 0; cur < colors.length; cur++) {
            if (colors[cur] == 0 &&
                    !dfs(cur, 1, colors, graph)) {
                return false;
            }
        }
        return true;
    }

    private boolean dfs(int cur, int curColor, int[] colors, int[][] graph) {
        colors[cur] = curColor;

        for (int neighbor : graph[cur]) {
            if (colors[neighbor] == 0 &&
                    !dfs(neighbor, 3 - curColor, colors, graph)) {
                return false;
            } else if (colors[neighbor] == colors[cur]) {
                return false;
            }
        }
        return true;
    }

}
